#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
API 数据模型
"""

from pydantic import BaseModel

class DownloadRequest(BaseModel):
    """下载请求模型"""
    url: str
    save_path: str = "./downloads"
    platform: str = "douyin"
    client_id: str = "default"

class CookieRequest(BaseModel):
    """Cookie请求模型"""
    cookies: str
    platform: str = "douyin"

class KuaishouBatchRequest(BaseModel):
    """快手批量下载请求模型"""
    user_id: str
    save_path: str = "./downloads"
    client_id: str = "default"
    task_id: str = None
